Pytorch different outputs between with transpose